package sf.tools;

import java.io.BufferedInputStream;
import java.io.BufferedOutputStream;
import java.io.Closeable;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.RandomAccessFile;
import java.nio.MappedByteBuffer;
import java.nio.channels.Channels;
import java.nio.channels.FileChannel;
import java.nio.channels.WritableByteChannel;
import java.util.zip.ZipEntry;
import java.util.zip.ZipInputStream;
import java.util.zip.ZipOutputStream;

/**
 * 本程序为liujiduo创建，wanglei修改 现在没发现什么中文问题，如果发现有乱码
 * 请参考-http://log-cd.iteye.com/blog/585647
 */
public final class ZipUtils {
    /**
     * 将filePaths 文件集合打包到zipPath 这个zip文件中
     * @param zipPath   压缩后的文件路径
     * @param filePaths 需要压缩的文件路径列表
     */
    public static void zip(String zipPath, String[] filePaths) {
        File target = new File(zipPath);
        // 压缩文件名=源文件名.zip
        // String zipName = zipPath;
        if (target.exists()) {
            target.delete(); // 删除旧的文件
        }
        FileOutputStream fos = null;
        ZipOutputStream zos = null;
        try {
            fos = new FileOutputStream(target);
            zos = new ZipOutputStream(new BufferedOutputStream(fos));
            // 添加对应的文件Entry
            addEntry(filePaths, zos);
        } catch (IOException e) {
            throw new RuntimeException(e);
        } finally {
            closeQuietly(zos, fos);
        }
    }

    /**
     * 扫描添加文件Entry
     * @param filePaths
     * @param zos       Zip文件输出流
     * @throws IOException
     */
    private static void addEntry(String[] filePaths, ZipOutputStream zos) throws IOException {
        FileInputStream fis = null;
        BufferedInputStream bis = null;
        File tempFile = null;
        try {

            for (String filePath : filePaths) {
                tempFile = new File(filePath);
                fis = new FileInputStream(tempFile);
                byte[] buffer = new byte[1024 * 10];
                bis = new BufferedInputStream(fis, buffer.length);
                int read = 0;
                zos.putNextEntry(new ZipEntry(tempFile.getName()));
                while ((read = bis.read(buffer, 0, buffer.length)) != -1) {
                    zos.write(buffer, 0, read);
                }
                zos.closeEntry();
            }
        } finally {
            closeQuietly(bis, fis);
        }
    }

    /**
     * 解压文件 -- 这个fun没有按照咱们的需要定制
     * @param filePath 压缩文件路径
     */
    public static void unzip(String filePath) {
        File source = new File(filePath);
        if (source.exists()) {
            ZipInputStream zis = null;
            BufferedOutputStream bos = null;
            try {
                zis = new ZipInputStream(new FileInputStream(source));
                ZipEntry entry = null;
                while ((entry = zis.getNextEntry()) != null) {
                    if (entry.isDirectory()) {
                        continue;
                    }
                    File target = new File(source.getParent(), entry.getName());
                    if (!target.getParentFile().exists()) {
                        // 创建文件父目录
                        target.getParentFile().mkdirs();
                    }
                    // 写入文件
                    bos = new BufferedOutputStream(new FileOutputStream(target));
                    int read = 0;
                    byte[] buffer = new byte[1024 * 10];
                    while ((read = zis.read(buffer, 0, buffer.length)) != -1) {
                        bos.write(buffer, 0, read);
                    }
                    bos.flush();
                }
                zis.closeEntry();
            } catch (IOException e) {
                throw new RuntimeException(e);
            } finally {
                closeQuietly(zis, bos);
            }
        }
    }

    public static void zipChannel(String zipPath, String[] filePaths) {
        File target = new File(zipPath);
        // 压缩文件名=源文件名.zip
        // String zipName = zipPath;
        if (target.exists()) {
            target.delete(); // 删除旧的文件
        }
        FileOutputStream fos = null;
        ZipOutputStream zos = null;
        WritableByteChannel writableByteChannel = null;
        try {
            fos = new FileOutputStream(target);
            zos = new ZipOutputStream(new BufferedOutputStream(fos));
            writableByteChannel = Channels.newChannel(zos);
            // 添加对应的文件Entry
            for (String filePath : filePaths) {
                File tempFile = new File(filePath);
                zos.putNextEntry(new ZipEntry(tempFile.getName()));
                FileChannel fileChannel = null;
                try {
                    fileChannel = new FileInputStream(tempFile).getChannel();
                    fileChannel.transferTo(0, fileChannel.size(), writableByteChannel);
                } finally {
                    zos.closeEntry();
                    closeQuietly(fileChannel);
                }
            }
        } catch (IOException e) {
            throw new RuntimeException(e);
        } finally {
            closeQuietly(zos, fos, writableByteChannel);
        }
    }

    public static void zipMap(String zipPath, String[] filePaths) {
        File target = new File(zipPath);
        // 压缩文件名=源文件名.zip
        // String zipName = zipPath;
        if (target.exists()) {
            target.delete(); // 删除旧的文件
        }
        FileOutputStream fos = null;
        ZipOutputStream zos = null;
        WritableByteChannel writableByteChannel = null;
        try {
            fos = new FileOutputStream(target);
            zos = new ZipOutputStream(new BufferedOutputStream(fos));
            writableByteChannel = Channels.newChannel(zos);
            // 添加对应的文件Entry
            for (String filePath : filePaths) {
                File tempFile = new File(filePath);
                zos.putNextEntry(new ZipEntry(tempFile.getName()));

                //内存中的映射文件
                MappedByteBuffer mappedByteBuffer = new RandomAccessFile(tempFile, "r").getChannel()
                        .map(FileChannel.MapMode.READ_ONLY, 0, tempFile.length());
                writableByteChannel.write(mappedByteBuffer);

                zos.closeEntry();
            }
        } catch (IOException e) {
            throw new RuntimeException(e);
        } finally {
            closeQuietly(zos, fos, writableByteChannel);
        }
    }

    /**
     * 关闭一个或多个流对象
     * @param closeables 可关闭的流对象列表
     * @throws IOException
     */
    public static void close(Closeable... closeables) throws IOException {
        if (closeables != null) {
            for (Closeable closeable : closeables) {
                if (closeable != null) {
                    closeable.close();
                }
            }
        }
    }

    /**
     * 关闭一个或多个流对象
     * @param closeables 可关闭的流对象列表
     */
    public static void closeQuietly(Closeable... closeables) {
        try {
            close(closeables);
        } catch (IOException e) {
            // do nothing
        }
    }

    public static void main(String[] args) {
        long l = System.currentTimeMillis();
        zipMap("d:/1.zip", new String[]{"D:\\SetupSoftware\\Navicat\\navicat150_premium_cs_x64.exe", "D:\\SetupSoftware\\Navicat\\navicat150_premium_cs_x86.exe"});
        System.out.println(System.currentTimeMillis() - l);
    }
}
